ABSTRACT
In this research work, a new deep learning model named VGG-COVIDNet has been proposed which can classify COVID-19 cases from normal cases over X-Rays and CT scan images of lungs. Medical practitioners use the X-Rays and CT scan images of lungs to identify whether a person is infected from COVID or not. In present times, it is very important to give real time COVID prediction with high reliability of results. Deep learning models equipped with machine learning support have been found very influential in accurate prediction of COVID or Non-COVID cases in real time. However, there are some limitations associated with the performance of these model which are model size, achieving good balance of model size and accuracy, and making a single model fitting well for both X-Ray and CT Scan image datasets. Keeping in mind these performance constraints, this new model (VGG-COVIDNet) has been proposed for real time prediction of COVID cases with good balance of model size and accuracy working well for both type of datasets (CT Scan and X-Ray). In order to control model size, an improved version of VGG-16 architecture has been proposed which contains only 13 convolutional layers and 5 fully connected layers. Multiple dropout layers have been added in the proposed architecture which can drop some percentage of features and applies random transformations to decrease the model over-fitting issue. Keeping in mind the primary goal to increase the model accuracy the proposed model has been trained on different datasets with ReLU activation function which is one of the best non-linear activation functions. Four different capacity datasets with CT scan and X-Ray images have been used to validate the performance of proposed model. The proposed model gives an overall accuracy of more than 90% on both types of input datasets i.e. X-Ray and CT Scan.
ABSTRACT
Chest X-ray (CXR) imaging is one of the most widely used and economical tests to diagnose a wide range of diseases. However, even for expert radiologists, it is a challenge to accurately diagnose diseases from CXR samples. Furthermore, there remains an acute shortage of trained radiologists worldwide. In the present study, a range of machine learning (ML), deep learning (DL), and transfer learning (TL) approaches have been evaluated to classify diseases in an openly available CXR image dataset. A combination of the synthetic minority over-sampling technique (SMOTE) and weighted class balancing is used to alleviate the effects of class imbalance. A hybrid Inception-ResNet-v2 transfer learning model coupled with data augmentation and image enhancement gives the best accuracy. The model is deployed in an edge environment using Amazon IoT Core to automate the task of disease detection in CXR images with three categories, namely pneumonia, COVID-19, and normal. Comparative analysis has been given in various metrics such as precision, recall, accuracy, AUC-ROC score, etc. The proposed technique gives an average accuracy of 98.66%. The accuracies of other TL models, namely SqueezeNet, VGG19, ResNet50, and MobileNetV2 are 97.33%, 91.66%, 90.33%, and 76.00%, respectively. Further, a DL model, trained from scratch, gives an accuracy of 92.43%. Two feature-based ML classification techniques, namely support vector machine with local binary pattern (SVM + LBP) and decision tree with histogram of oriented gradients (DT + HOG) yield an accuracy of 87.98% and 86.87%, respectively.